import matplotlib.pyplot as plt

#x_values = [1, 2, 3, 4, 5]
#y_values = [1, 4, 9, 16, 25]
# 自动计算数据
x_values = range(1, 1001)
y_values = [x**2 for x in x_values]

plt.style.use("seaborn-v0_8-dark")
fig, ax = plt.subplots()
#ax.scatter(x_values, y_values, s=10)
#ax.scatter(x_values, y_values, c='red', s=10)
# 使用颜色映射 
ax.scatter(x_values, y_values, c=y_values, cmap=plt.cm.Reds, s=10)

# 设置图表标题并给坐标轴加上标签
ax.set_title("Square Number", fontsize=24)
ax.set_xlabel("Value", fontsize=14)
ax.set_ylabel("Square Of Value", fontsize=14)

# 设置坐标轴刻度大小
ax.tick_params(axis='both', which='major', labelsize=14)

# 设置每个坐标轴的取值范围
ax.axis([0, 1100, 0, 1100000])

# 保存图表
plt.savefig('squares_plot.png', bbox_inches='tight')

plt.show()